CUDA_VISIBLE_DEVICES=0 OMP_NUM_THREADS=1 python inference.py
